from enum import Enum
from sklearn import svm
from sklearn.ensemble import RandomForestClassifier
import xgboost as xgb
import lightgbm as lgb


class Model(Enum):
    SVM = svm.SVC(C=1, gamma=1, kernel='linear',probability=True,random_state=2)
    RandomForest = RandomForestClassifier(random_state=2)
    XGBoost = xgb.XGBClassifier(random_state=2,eta=0.1,max_depth=10)
    LightGBM = lgb.LGBMClassifier(random_state=2)
